package com.gimi.test_jpa.com.gimi.mq;

import lombok.extern.slf4j.Slf4j;
import org.apache.rocketmq.client.producer.SendCallback;
import org.apache.rocketmq.client.producer.SendResult;
import org.apache.rocketmq.spring.core.RocketMQTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.messaging.support.MessageBuilder;
import org.springframework.stereotype.Component;

@Component
@Slf4j
public class RocketMQProducer{

    @Autowired
    private RocketMQTemplate rocketMQTemplate;

    @Value("${rocketmq.producer.send-message-timeout}")
    private Integer messageTimeOut;

    /**
     * 发送普通消息
     */
    public void sendMsg(String topic,String msgBody){
        rocketMQTemplate.syncSend(topic, MessageBuilder.withPayload(msgBody).build());
    }

    /**
     * 发送异步消息 在SendCallback中可处理相关成功失败时的逻辑
     */
    public void sendAsyncMsg(String topic,String msgBody){
        rocketMQTemplate.asyncSend(topic,MessageBuilder.withPayload(msgBody).build(), new SendCallback() {
            @Override
            public void onSuccess(SendResult sendResult) {
                // 处理消息发送成功逻辑
                log.info(sendResult.toString());
            }
            @Override
            public void onException(Throwable e) {
                // 处理消息发送异常逻辑
            }
        });
    }

    /**
     * 发送延时消息<br/>
     * 在start版本中 延时消息一共分为18个等级分别为：1s 5s 10s 30s 1m 2m 3m 4m 5m 6m 7m 8m 9m 10m 20m 30m 1h 2h<br/>
     */
    public void sendDelayMsg(String topic,String msgBody, Integer delayLevel){
        rocketMQTemplate.syncSend(topic,MessageBuilder.withPayload(msgBody).build(),messageTimeOut,delayLevel);
    }

    /**
     * 发送带tag的消息,直接在topic后面加上":tag"
     */
    public void sendTagMsg(String topic,String tag,String msgBody){
        rocketMQTemplate.syncSend(topic+":"+tag,MessageBuilder.withPayload(msgBody).build());
    }

    /**
     * 发送顺序信息
     *
     *hashKey: 为了保证报到同一个队列中，将消息发送到orderTopic主题上
     * (他的作用是会根据他的hash值计算发送到哪一个队列，我用的是同一个值,那么他们的hash一样就可以保证发送到同一个队列里)
     */
    public void sendOrderly(String topic,String msgBody,String hashKey){
        rocketMQTemplate.syncSendOrderly(topic,MessageBuilder.withPayload(msgBody).build(),hashKey);
    }
}
